import os
import numpy as np

def load_dataset(dataset_name):
    """
    加载指定的数据集
    格式：CSV文件，每行为 x坐标,y坐标,类标签
    """
    dataset_path = os.path.join('人工数据集', f'{dataset_name}.csv')
    
    try:
        # 加载CSV文件
        data = np.loadtxt(dataset_path, delimiter=',')
        
        # 前两列作为特征 (x,y坐标)
        X = data[:, :2]
        
        # 最后一列为类标签
        y = data[:, -1]
        
        print(f"成功加载数据集: {dataset_name}, 形状: {X.shape}")
        return X, y
            
    except Exception as e:
        print(f"加载数据集失败: {e}")
        return None, None